from VTActor import VTActor, VTActorRenderType


class VTLinesActor(VTActor):
    def __init__(self, poly_data):
        super().__init__(poly_data, {
            "defaultCellColor": [192, 56, 255, 255]
        })
        self.set_render_type(VTActorRenderType.WIREFRAME)
        self.set_width(2)

    def set_width(self, width):
        self.instance.get_property().set_line_width(width)
